package net.cmkj.mine.util.tpp.coder;

import com.jfinal.kit.PathKit;
import com.jfinal.kit.PropKit;
import com.jfinal.plugin.activerecord.generator.Generator;
import com.jfinal.plugin.druid.DruidPlugin;

import net.cmkj.mine.conf.MineConfig;

import javax.sql.DataSource;


/**
 * model代码生成器
 * @author zzcmkj
 *
 */
public class CodeGenerator {
	
	private static Generator generator = null;
	private static WMetaBuilder metaBuilder = null;
	
	public static DataSource getDataSource() {
		PropKit.use("db_config.txt");
		DruidPlugin druidPlugin = MineConfig.createDruidPlugin();
		druidPlugin.set(1, 1, 2);
		druidPlugin.start();
		return druidPlugin.getDataSource();
	}
	
	public static void generate(String tableName) {
		// base info 所使用的包名
		String baseModelPackageName = "net.cmkj.mine.bean.base";
		// base info 文件保存路径
		String modelOutputDir = PathKit.getWebRootPath() + "/src/main/java/net/cmkj/mine/bean";
		String baseModelOutputDir = modelOutputDir + "/base";
		String mappingOutputDir = modelOutputDir + "/coder";
		
		// info 所使用的包名 (MappingKit 默认使用的包名)
		String modelPackageName = "net.cmkj.mine.bean";
		// info 文件保存路径 (MappingKit 与 DataDictionary 文件默认保存路径)
		//String modelOutputDir = baseModelOutputDir + "/..";
		
		// 创建生成器
		if (generator == null) {
			generator = new Generator(getDataSource(), baseModelPackageName, baseModelOutputDir, modelPackageName, modelOutputDir);
			generator.setBaseModelTemplate("/base_model_template.jf");
			generator.setModelTemplate("/model_template.jf");
			// 设置是否生成链式 setter 方法
			generator.setGenerateChainSetter(false);
			// 添加不需要生成的表名
			generator.addExcludedTable("");
			
			if (metaBuilder == null) {
				metaBuilder = new WMetaBuilder(getDataSource());
			}
		}
	    metaBuilder.setCreateTableName(tableName);
		generator.setMetaBuilder(metaBuilder);
		
		WMappingKitGenerator mappingKitBulid = new WMappingKitGenerator(modelPackageName+".coder", mappingOutputDir);
		generator.setMappingKitGenerator(mappingKitBulid);
		
		// 设置是否在 Model 中生成 dao 对象
		generator.setGenerateDaoInModel(true);
		// 设置是否生成链式 setter 方法
		generator.setGenerateChainSetter(true);
		// 设置是否生成字典文件
		generator.setGenerateDataDictionary(false);
		// 设置需要被移除的表名前缀用于生成modelName。例如表名 "osc_user"，移除前缀 "osc_"后生成的model名为
		// "User"而非 OscUser
		generator.setRemovedTableNamePrefixes("t_");
		// 生成
		generator.generate();
	}
	
	public static void main(String[] args) {
		generate("columns");
	}

}
